Journals
  Publication Years
  Keywords
Search within results Open Search
Please wait a minute...
For Selected: Toggle Thumbnails
Runoff forecast model based on graph attention network and dual-stage attention mechanism
Hexuan HU, Huachao SUI, Qiang HU, Ye ZHANG, Zhenyun HU, Nengwu MA
Journal of Computer Applications    2022, 42 (5): 1607-1615.   DOI: 10.11772/j.issn.1001-9081.2021050829
Abstract606)   HTML11)    PDF (2505KB)(170)       Save

To improve the accuracy of watershed runoff volume prediction, and considering the lack of model transparency and physical interpretability of data-driven hydrological model, a new runoff forecast model named Graph Attention neTwork and Dual-stage Attention mechanism-based Long Short-Term Memory network (GAT-DALSTM) was proposed. Firstly, based on the hydrological data of watershed stations, graph neural network was introduced to extract the topology of watershed stations and generate the feature vectors. Secondly, according to the characteristics of hydrological time series data, a runoff forecast model based on dual-stage attention mechanism was established to predict the watershed runoff volume, and the reliability and transparency of the proposed model were verified by the model evaluation method based on attention coefficient heat map. On the Tunxi watershed dataset, the proposed model was compared with Graph Convolution Neural network (GCN) and Long Short-Term Memory network (LSTM) under each prediction step. Experimental results show that, the Nash-Sutcliffe efficiency coefficient of the proposed model is increased by 3.7% and 4.9% on average respectively, which verifies the accuracy of GAT-DALSTM runoff forecast model. By analyzing the heat map of attention coefficient from the perspectives of hydrology and application, the reliability and practicability of the proposed model were verified. The proposed model can provide technical support for improving the prediction accuracy and model transparency of watershed runoff volume.

Table and Figures | Reference | Related Articles | Metrics
Image super-resolution reconstruction network based on multi-channel attention mechanism
Ye ZHANG, Rong LIU, Ming LIU, Ming CHEN
Journal of Computer Applications    2022, 42 (5): 1563-1569.   DOI: 10.11772/j.issn.1001-9081.2021030498
Abstract273)   HTML5)    PDF (3016KB)(120)       Save

The existing image super-resolution reconstruction methods are affected by texture distortion and details blurring of generated images. To address these problems, a new image super-resolution reconstruction network based on multi-channel attention mechanism was proposed. Firstly, in the texture extraction module of the proposed network, a multi-channel attention mechanism was designed to realize the cross-channel information interaction by combining one-dimensional convolution, thereby achieving the purpose of paying attention to important feature information. Then, in the texture recovery module of the proposed network, the dense residual blocks were introduced to recover part of high-frequency texture details as many as possible to improve the performance of model and generate high-quality reconstructed images. The proposed network is able to improve visual effects of reconstructed images effectively. Besides, the results on benchmark dataset CUFED5 show that the proposed network has achieved the 1.76 dB and 0.062 higher in Peak Signal-to-Noise Ratio (PSNR) and Structural SIMilarity (SSIM) compared with the classic Super-Resolution using Convolutional Neural Network (SRCNN) method. Experimental results show that the proposed network can increase the accuracy of texture migration, and effectively improve the quality of generated images.

Table and Figures | Reference | Related Articles | Metrics